DefaultFailureHandlerTest.java

package com.google.android.apps.common.testing.ui.espresso.base;

import static com.google.android.apps.common.testing.ui.espresso.Espresso.onView;
import static com.google.android.apps.common.testing.ui.espresso.assertion.ViewAssertions.matches;
import static com.google.android.apps.common.testing.ui.espresso.matcher.ViewMatchers.isDisplayed;
import static com.google.android.apps.common.testing.ui.espresso.matcher.ViewMatchers.isRoot;
import static com.google.common.base.Throwables.getStackTraceAsString;
import static org.hamcrest.Matchers.not;

import com.google.android.apps.common.testing.ui.espresso.AmbiguousViewMatcherException;
import com.google.android.apps.common.testing.ui.espresso.NoMatchingViewException;
import com.google.android.apps.common.testing.ui.espresso.ViewAssertion;
import com.google.android.apps.common.testing.ui.testapp.MainActivity;
import com.google.common.base.Optional;

import android.test.ActivityInstrumentationTestCase2;
import android.view.View;

import junit.framework.AssertionFailedError;

import org.hamcrest.BaseMatcher;
import org.hamcrest.Description;
import org.hamcrest.Matcher;

/**
 * Tests Espresso's default failure handling.
 */
public class DefaultFailureHandlerTest extends ActivityInstrumentationTestCase2<MainActivity> {

  @SuppressWarnings("deprecation")
  public DefaultFailureHandlerTest() {
    // Supporting froyo.
    super("com.google.android.apps.common.testing.ui.testapp", MainActivity.class);
  }

  @Override
  public void setUp() throws Exception {
    super.setUp();
    getActivity();
  }

  public void testMismatchInCheck() {
    try {
      onView(isRoot()).check(matches(not(isDisplayed())));
      fail("Previous call expected to fail");
    } catch (AssertionFailedError e) {
      assertFailureStackContainsThisClass(e);
    }
  }

  public void testCustomAssertionError() {
    try {
      onView(isRoot()).check(new ViewAssertion() {
        @Override
        public void check(
            Optional<View> view, Optional<NoMatchingViewException> noViewFoundException) {
          assertFalse(true);
        }
      });
      fail("Previous call expected to fail");
    } catch (AssertionFailedError e) {
      assertFailureStackContainsThisClass(e);
    }
  }

  public void testNoMatchingViewException() {
    try {
      onView(withMatchesThatReturns(false)).check(matches(not(isDisplayed())));
      fail("Previous call expected to fail");
    } catch (NoMatchingViewException e) {
      assertFailureStackContainsThisClass(e);
    }
  }

  public void testAmbiguousViewMatcherException() {
    try {
      onView(withMatchesThatReturns(true)).check(matches(isDisplayed()));
    } catch (RuntimeException e) {
      assertTrue(e instanceof AmbiguousViewMatcherException);
      assertFailureStackContainsThisClass(e);
    }
  }

  private void assertFailureStackContainsThisClass(Throwable e) {
    assertTrue(getStackTraceAsString(e).contains(getClass().getSimpleName().toString()));
  }

  private static Matcher<View> withMatchesThatReturns(final boolean returnValue) {
    return new BaseMatcher<View>() {
      @Override
      public void describeTo(Description description) {
        description.appendText("matches=" + returnValue);
      }

      @Override
      public boolean matches(Object item) {
        return returnValue;
      }
    };
  }

}